import pandas as pd
x = input("请输入待查找的pd_id:")
target_pd_id = x
chunk_size = 498
df = pd.read_csv('/root/SHAI_MXZ5/CrossBorderAIChallenge/input/platform-policy-compliance/answer_sheet.csv',chunksize = chunk_size)
target_list = []
for chunk in df:
    mask = chunk['pd_id'] == target_pd_id
    # 使用布尔掩码筛选匹配的行
    matched_rows_in_chunk = chunk[mask]
    # 如果找到了匹配的行，追加到 target_list 列表中
    if not matched_rows_in_chunk.empty:
        target_list.append(matched_rows_in_chunk)
print(target_list)
    
